cs304 assignment 1 2022 solution